import org.junit.Before;
import org.junit.Test;

import junit.framework.TestCase;



public class TesteTempoEmprestimo extends TestCase {
	TempoEmprestimo tempoEmprestimoLivro;
	
	@Before
	public void setUp() {
		tempoEmprestimoLivro = new TempoEmprestimo(21, 7, 7, 2);
	}
	
	@Test
	public void testGetTempoEmprestimo() throws Exception {
		assertEquals("Erro de getTempoProfessor para livro", 21, tempoEmprestimoLivro.getTempo("Professor"));
		assertEquals("Erro de getTempoAluno para livro", 7, tempoEmprestimoLivro.getTempo("Aluno"));
		assertEquals("Erro de getTempoFuncionario para livro", 7, tempoEmprestimoLivro.getTempo("Funcionario"));
		assertEquals("Erro de getTempoOutro para livro", 2, tempoEmprestimoLivro.getTempo("Outro"));
	}
	
	@Test
	public void testSetTempoEmprestimo() throws Exception {
		tempoEmprestimoLivro.setTempo("Professor", 10);
		assertEquals("Erro de setTempoProfessor para livro", 10, tempoEmprestimoLivro.getTempo("Professor"));
		tempoEmprestimoLivro.setTempo("Aluno", 12);
		assertEquals("Erro de setTempoAluno para livro", 12, tempoEmprestimoLivro.getTempo("Aluno"));
		tempoEmprestimoLivro.setTempo("Funcionario", 5);
		assertEquals("Erro de setTempoFuncionario para livro", 5, tempoEmprestimoLivro.getTempo("Funcionario"));
		tempoEmprestimoLivro.setTempo("Outro", 4);
		assertEquals("Erro de setTempoOutro para livro", 4, tempoEmprestimoLivro.getTempo("Outro"));
		
	}
	
}
